from extension import db
from datetime import date


class Module(db.Model):  # 表名将会是 model（自动生成，小写处理）
    id = db.Column(db.Integer, primary_key=True)  # 主键
    module = db.Column(db.String(20))  # 小组
    def save(self):
        db.session.add(self)
        db.session.commit()
        print('Module save Done.')


class Equipment(db.Model):  # 表名将会是 equipment（自动生成，小写处理）
    id = db.Column(db.Integer, primary_key=True)  # 主键
    module = db.Column(db.String(20))  # 小组名
    name = db.Column(db.String(20))  # 设备名
    create_time = db.Column(db.Date, default=date.today())
    value = db.Column(db.Integer)  # 报警次数
    def save(self):
        db.session.add(self)
        db.session.commit()
        print('Equipment save Done.')
    